#include "binary_tree.h"

TreeNode *root{}; // 根节点

/* 查找节点 */
TreeNode *search(int num)
{
    TreeNode *cur = root;
    // 循环查找，越过叶节点后跳出
    while (cur != nullptr)
    {
        // 目标节点在 cur 的右子树中
        if (cur->val < num)
            cur = cur->right;
        // 目标节点在 cur 的左子树中
        else if (cur->val > num)
            cur = cur->left;
        // 找到目标节点，跳出循环
        else
            break;
    }
    // 返回目标节点
    return cur;
}